import unittest
from src20211230 import testbaidu2
from src20211230 import testbaidu1
import unittest

def creatSuit():
    # 要把不同的测试脚本的类中的需要执行的方法放在一个测试套件中


   #如果我需要把一个测试脚本中年所有的测试用例都添加到suit中，怎么做？
   # makeSuit
    suit = unittest.TestSuite()
    suit.addTest(unittest.makeSuite(testbaidu1.Baidu1))
    suit.addTest(unittest.makeSuite(testbaidu2.Baidu2))
    return suit
#     TestLoader
    suit1 = unittest.TestLoader().loadTestsFromTestCase(testbaidu1.Baidu1)
    suit2 = unittest.TestLoader().loadTestsFromTestCase(testbaidu2.Baidu2)
    suit = unittest.TestSuite([suit1, suit2])
    return suit
# 可以把一个文件夹下面所有的测试脚本中的测试用例放入测试套件
    discover = unittest.defaultTestLoader.discover("../src20211230", pattern="testbaidu*.py", top_level_dir=None)
    return discover
if __name__ == "__main__":
    suit = creatSuit()
    # verbersity= 0, 1, 2
    runner = unittest.TextTestRunner(verbosity=2)
    runner.run(suit)